// 多源 bfs 问题
// 把多个源点当成一个超级源点 - 即将所有源点加入队列中，之后进行 bfs 搜索
// 正难则反

// 例题 1：
//给定一个由 0 和 1 组成的矩阵 mat ，请输出一个大小相同的矩阵，其中每一个格子是 mat 中对应位置元素到最近的 0 的距离。
//
//        两个相邻元素间的距离为 1 。
//
//        示例 1：
//
//        输入：mat = [[0,0,0],[0,1,0],[0,0,0]]
//        输出：[[0,0,0],[0,1,0],[0,0,0]]
//        示例 2：
//
//        输入：mat = [[0,0,0],[0,1,0],[1,1,1]]
//        输出：[[0,0,0],[0,1,0],[1,2,1]]
//
//
//        提示：
//
//        m == mat.length
//        n == mat[i].length
//        1 <= m, n <= 104
//        1 <= m * n <= 104
//        mat[i][j] is either 0 or 1.
//        mat 中至少有一个 0

// 解题思路：
// 正难则反 - 将所有的 0 的坐标加入队列
// 进行多源 bfs 搜索，扩展到的位置标上距离
// 可以通过将 ret[i][j] 初始化成 -1，用来代替 check[i][j] 进行标记
// 距离也可以通过 ret[x][y] = ret[a][b] + 1 计算

import java.util.LinkedList;
import java.util.Queue;

public class UpdateMatrix {
    public int[][] updateMatrix(int[][] mat) {
        int m = mat.length;
        int n = mat[0].length;
        int[] dx = {0, 0, 1, -1};
        int[] dy = {1, -1, 0, 0};
        int[][] ret = new int[m][n];
        Queue<int[]> queue = new LinkedList<>();
        for(int i = 0; i < m; i++){
            for(int j = 0; j < n; j++){
                if(mat[i][j] == 0){
                    queue.offer(new int[] {i, j});
                    ret[i][j] = 0;
                }else ret[i][j] = -1;
            }
        }
        while(!queue.isEmpty()){
            int[] point = queue.poll();
            int a = point[0];
            int b = point[1];
            for(int l = 0; l < 4; l++){
                int x = a + dx[l];
                int y = b + dy[l];
                if(x >= 0 && x < m && y >= 0 && y < n && mat[x][y] != 0 && ret[x][y] == -1){
                    ret[x][y] = ret[a][b] + 1;
                    queue.offer(new int[] {x, y});
                }
            }
        }
        return ret;
    }
}
